from ultralytics import YOLO

# 初始化一个模型
model = YOLO(r'./yolov8n.pt')  # load a pretrained model (recommended for training)

if __name__ == '__main__':
    # Train the model
    # 模型训练命令
    model.train(data=r"D:/E-competition/AI/yolov8-test/mydata.yaml",epochs=500, imgsz=640, batch=16)

    # 参数说明:
    # 1.data: 数据配置文件 （需要修改为实际的mydata.yaml文件绝对路径）
    # 2.epochs: 训练轮次 （训练完一遍之后，如果R值中有低于0.5的将epochs为100重新训练）
    # 3.imgsz: 输入图片大小
    # 4.batch: 批次大小 （根据显卡型号去决定，如果是30系列的可以设置为16，如果是10系列则设置为8或4）